package com.googlecode.mailnet.emf.netmodel.visualization.failureprobability;

import com.googlecode.mailnet.emf.netmodel.visualization.ModelDecorator;
import com.googlecode.mailnet.emf.netmodel.visualization.ShapeDecorator;
import com.googlecode.mailnet.info.ILinkInfo;
import com.googlecode.mailnet.info.INodeInfo;

public class FailureProbabilityModelDecorator extends ModelDecorator {

	@Override
	public ShapeDecorator getLinkDecorator(ILinkInfo linkInfo) {
		return new FailureProbabilityLinkDecorator(linkInfo, context);
	}

	@Override
	public ShapeDecorator getNodeDecorator(INodeInfo nodeInfo) {
		return new FailureProbabilityNodeDecorator(nodeInfo, context);
	}

}
